Separating Thinking from Spelling for Efficient Generative Recommendation
分离思考与拼写的高效生成式推荐
🧭 生成式推荐 · Semantic ID 推理优化 ★ 深度解读
将生成式推荐中 backbone 的历史建模与 SID 逐码生成解耦,由轻量 decoder 完成 beam search,实现 3-8 倍吞吐提升且精度不降
📌 工作参考价值(8 点)
- 架构选择:重 encoder + 轻 decoder 分离 serving
- 在生成式推荐(Semantic ID)线上推理中,让大 backbone 只跑一次 prefill 输出固定大小的 next-item concept(如 1KiB),beam search 全部下沉到 2 层 256 宽的轻量 decoder。concept 可作为 per-user state 预计算并缓存,在线只有轻 decoder 参与 decode,避免大模型反复处理 O(BK) 个 decode token。
- 输入侧融合 item SID tokens 降低 prefill
- 用 MLP 把每个 item 的 K 个 code embedding 融合成单个 token,历史长度从 TK 降到 T;训练时通过 block-causal mask 为每个位置提供下一 item 的 gold prefix targets,实现 dense training,一次前向即可监督所有 transitions。
- 轻 decoder 训练用教师分布做 label smoothing
- 将 full-backbone teacher 在 gold prefix 上的 next-code 分布与 one-hot 混合(α 为 KD 权重),作为 decoder 的 CE 目标。在小数据集上可带来 16-23% NDCG@10 提升,且 teacher 仅训练时使用,不增加 serving 成本。
- 复购/重复消费场景加 history pointer
- 对音乐、视频、电商复购等 repeat-heavy 领域,基于 backbone last state 对历史 item 的 attention 构造 pointer,与生成分布混合(gate 由 last state 学习)。在 Yambda 数据集上,pointer 将 Deco NDCG@10 从 0.079 提升到 0.171,几乎追平 FlatGR。